#!/usr/bin/env python
"""
创建测试Excel文件
"""
import pandas as pd
import os

def create_test_excel_file():
    """创建测试用的Excel文件"""
    # 创建测试数据
    data = {
        '序号': [1, 2, 3, 4, 5],
        '投诉人': ['张三', '李四', '王五', '赵六', '钱七'],
        '被投诉方': ['A公司', 'B部门', 'C单位', 'D机构', 'E组织'],
        '投诉内容': [
            '服务质量差，态度不好',
            '处理效率低，拖延时间',
            '收费不合理，存在乱收费',
            '信息不透明，缺乏沟通',
            '设施不完善，环境差'
        ],
        '投诉地址': ['北京市朝阳区', '上海市浦东新区', '广州市天河区', '深圳市南山区', '杭州市西湖区'],
        '投诉时间': ['2024-01-01', '2024-01-02', '2024-01-03', '2024-01-04', '2024-01-05'],
        '处理状态': ['待处理', '处理中', '已解决', '已解决', '待处理']
    }
    
    # 创建DataFrame
    df = pd.DataFrame(data)
    
    # 确保upload目录存在
    upload_dir = 'upload'
    if not os.path.exists(upload_dir):
        os.makedirs(upload_dir)
    
    # 保存为Excel文件
    excel_file = os.path.join(upload_dir, 'test_complaints.xlsx')
    df.to_excel(excel_file, index=False, sheet_name='投诉数据')
    
    print(f"测试Excel文件已创建: {excel_file}")
    print(f"文件大小: {os.path.getsize(excel_file)} bytes")
    print(f"数据行数: {len(df)}")
    print(f"数据列数: {len(df.columns)}")
    
    return excel_file

if __name__ == "__main__":
    create_test_excel_file()